home *** CD-ROM | disk | FTP | other *** search
/ MacHack 2000 / MacHack 2000.toast / pc / The Hacks / MacHacksBug / Python 1.5.2c1 / Extensions / Imaging / Scripts / gifmaker.py < prev    next >
Encoding:
Python Source  |  2000-06-23  |  2.7 KB  |  136 lines
  1. #
  2. # The Python Imaging Library
  3. # $Id: gifmaker.py,v 1.1.1.1 1998/08/18 13:07:59 sjoerd Exp $
  4. #
  5. # convert sequence format to GIF animation
  6. #
  7. # history:
  8. #    97-01-03 fl    created
  9. #
  10. # Copyright (c) Secret Labs AB 1997.  All rights reserved.
  11. # Copyright (c) Fredrik Lundh 1997.
  12. #
  13. # See the README file for information on usage and redistribution.
  14. #
  15.  
  16. #
  17. # For special purposes, you can import this module and call
  18. # the makedelta or compress functions yourself.  For example,
  19. # if you have an application that generates a sequence of
  20. # images, you can convert it to a GIF animation using some-
  21. # thing like the following code:
  22. #
  23. #    import Image
  24. #    import gifmaker
  25. #    
  26. #    sequence = []
  27. #    
  28. #    # generate sequence
  29. #    for i in range(100):
  30. #        im = <generate image i>
  31. #        sequence.append(im)
  32. #
  33. #    # write GIF animation
  34. #    fp = open("out.gif", "wb")
  35. #    gifmaker.makedelta(fp, sequence)
  36. #    fp.close()
  37. # 
  38. # Alternatively, use an iterator to generate the sequence, and
  39. # write data directly to a socket.  Or something...
  40. #
  41.  
  42. import Image, ImageChops
  43. import string
  44.  
  45. from GifImagePlugin import getheader, getdata
  46.  
  47. # --------------------------------------------------------------------
  48. # sequence iterator
  49.  
  50. class image_sequence:
  51.     def __init__(self, im):
  52.     self.im = im
  53.     def __getitem__(self, ix):
  54.     try:
  55.         if ix:
  56.         self.im.seek(ix)
  57.         return self.im
  58.     except EOFError:
  59.         raise IndexError # end of sequence
  60.  
  61. # --------------------------------------------------------------------
  62. # straightforward delta encoding
  63.  
  64. def makedelta(fp, sequence):
  65.     """Convert list of image frames to a GIF animation file"""
  66.  
  67.     frames = 0
  68.  
  69.     previous = None
  70.  
  71.     for im in sequence:
  72.  
  73.     #
  74.     # FIXME: write graphics control block before each frame
  75.  
  76.     if not previous:
  77.  
  78.         # global header
  79.         for s in getheader(im) + getdata(im):
  80.         fp.write(s)
  81.  
  82.     else:
  83.  
  84.         # delta frame
  85.         delta = ImageChops.subtract_modulo(im, previous)
  86.  
  87.         bbox = delta.getbbox()
  88.  
  89.         if bbox:
  90.  
  91.         # compress difference
  92.         for s in getdata(im.crop(bbox), offset = bbox[:2]):
  93.             fp.write(s)
  94.  
  95.         else:
  96.         # FIXME: what should we do in this case?
  97.         pass
  98.  
  99.     previous = im.copy()
  100.  
  101.     frames = frames + 1
  102.  
  103.     fp.write(";")
  104.  
  105.     return frames
  106.  
  107. # --------------------------------------------------------------------
  108. # main hack
  109.  
  110. def compress(infile, outfile):
  111.  
  112.     # open input image, and force loading of first frame
  113.     im = Image.open(infile)
  114.     im.load()
  115.  
  116.     # open output file
  117.     fp = open(outfile, "wb")
  118.  
  119.     seq = image_sequence(im)
  120.  
  121.     makedelta(fp, seq)
  122.  
  123.     fp.close()
  124.  
  125.  
  126. if __name__ == "__main__":
  127.  
  128.     import sys
  129.  
  130.     if len(sys.argv) < 3:
  131.     print "GIFMAKER -- create GIF animations"
  132.     print "Usage: gifmaker infile outfile"
  133.     sys.exit(1)
  134.  
  135.     compress(sys.argv[1], sys.argv[2])
  136.